package com.boilermaker.mq.mqserver.core;

import com.boilermaker.mq.common.ConsumerEnv;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.Getter;
import lombok.Setter;

import java.sql.SQLException;
import java.util.*;
import java.util.concurrent.atomic.AtomicInteger;

/*
 * 表示一个存储消息的队列
 */
@Getter
@Setter
public class MSGQueue {
    // 唯一身份标识
    private String name;
    // 持久化储存
    private boolean durable = false;
    // 该队列是否为独占队列（只能由一个消费者使用）
    private boolean exclusive = false; // TODO:自行实现
    // 当前队列若无消费者使用，是否自动删除
    private boolean autoDelete = false; // TODO:自行实现
    // 其他选项或参数（为方便存入数据库，应当转换为 json）
    private Map<String, Object> arguments = new HashMap<>(); // TODO:自行实现

    // 当前队列被哪些消费者订阅
    private List<ConsumerEnv> consumerList = new ArrayList<>();
    // 辅助轮询时计算待取元素下标，使用原子类保证线程安全
    private AtomicInteger consumerSeq = new AtomicInteger(0);

    // 添加消费者
    public void addConsumer(ConsumerEnv consumerEnv) {
        synchronized (this) {
            consumerList.add(consumerEnv);
        }
    }

    // 以轮询的方式选择一个消费者去消费消息
    public ConsumerEnv chooseConsumer() {
        if (consumerList.isEmpty()) return null;
        int index = consumerSeq.get() % consumerList.size();
        consumerSeq.getAndIncrement();
        return consumerList.get(index);
    }

    // Mybatis 对于 Java 的基本数据类型、字符串和日期时间类型能够自动转换，但不会对集合类进行自动转换
    // 手动转换为 JSON：Mybatis 写数据库前，会先调用属性的 get 方法，因此在 get 中预设转换的逻辑
    public String getArguments() {
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            return objectMapper.writeValueAsString(arguments);
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
        return "{}";
    }

    // Mybatis 从数据库中读完数据，会调用这个方法将数据写入 Java 对象
    public void setArguments(String argumentsJson) {
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            this.arguments = objectMapper.readValue(argumentsJson, new TypeReference<
                    HashMap<String, Object>>() {
            });
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
    }

    public void setArguments(Map<String, Object> arguments) {
        this.arguments = arguments;
    }
}
